#############################################
# Image Style Transfer Scripts for SandDust #
# DataGenerateToolGroup@JSZN #
#############################################

import cv2
import argparse
from PIL import Image
from algorithms import SandDustInferencer

def main():
	parser = argparse.ArgumentParser(description='Scripts for Image SandDust')
	parser.add_argument('-i', '--image', default='/home/song/Codes/refind_algorithm_library/data/images/01.jpg', type=str, 
					 help='image path for inference')
	args = parser.parse_args()

	input = cv2.imread(args.image)
	results = SandDustInferencer().inference(input)
	
	for result in results:
		result = Image.fromarray(result)
		result.show()


if __name__ == '__main__':
	main()

